# 模型的加载
import torch
import torchvision.models
# 方式 1 加载数据
vgg = torch.load('./models/vgg_method1.pth', weights_only=False)
print(vgg)

# 方式 2 加载数据
vgg2 = torch.load('./models/vgg_method2.pth', weights_only=False)
print(vgg2)
# 将方式 2 加载的权重转换为模型结构进行输出
vgg2 = torchvision.models.vgg16()
vgg2.load_state_dict(torch.load('./models/vgg_method2.pth'))